/*
 *  Copyright (C) 2010-2014 JPEXS
 * 
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 * 
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 * 
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.jpexs.decompiler.flash.abc.types;

import com.jpexs.decompiler.flash.abc.avm2.ConstantPool;
import com.jpexs.helpers.Helper;
import java.util.List;

public class Multiname {

    public static final int QNAME = 7;
    public static final int QNAMEA = 0x0d;
    public static final int RTQNAME = 0x0f;
    public static final int RTQNAMEA = 0x10;
    public static final int RTQNAMEL = 0x11;
    public static final int RTQNAMELA = 0x12;
    public static final int MULTINAME = 0x09;
    public static final int MULTINAMEA = 0x0e;
    public static final int MULTINAMEL = 0x1b;
    public static final int MULTINAMELA = 0x1c;
    public static final int TYPENAME = 0x1d;
    private static final int[] multinameKinds = new int[]{QNAME, QNAMEA, MULTINAME, MULTINAMEA, RTQNAME, RTQNAMEA, MULTINAMEL, RTQNAMEL, RTQNAMELA, MULTINAMELA, TYPENAME};
    private static final String[] multinameKindNames = new String[]{"Qname", "QnameA", "Multiname", "MultinameA", "RTQname", "RTQnameA", "MultinameL", "RTQnameL", "RTQnameLA", "MultinameLA", "TypeName"};
    public int kind = -1;
    public int name_index = -1;
    public int namespace_index = -1;
    public int namespace_set_index = -1;
    public int qname_index = -1; //for TypeName
    public List<Integer> params; //for TypeName

    public boolean validType() {
        boolean cnt = false;
        for (int i = 0; i < multinameKinds.length; i++) {
            if (multinameKinds[i] == kind) {
                cnt = true;
            }
        }
        return cnt;
    }

    public Multiname(int kind, int name_index, int namespace_index, int namespace_set_index, int qname_index, List<Integer> params) {
        this.kind = kind;
        this.name_index = name_index;
        this.namespace_index = namespace_index;
        this.namespace_set_index = namespace_set_index;
        this.qname_index = qname_index;
        this.params = params;
        if (!validType()) {
            throw new RuntimeException("Invalid multiname kind:" + kind);
        }
    }

    public boolean isAttribute() {
        if (kind == QNAMEA) {
            return true;
        }
        if (kind == MULTINAMEA) {
            return true;
        }
        if (kind == RTQNAMEA) {
            return true;
        }
        if (kind == RTQNAMELA) {
            return true;
        }
        if (kind == MULTINAMELA) {
            return true;
        }
        return false;
    }

    public boolean isRuntime() {
        if (kind == RTQNAME) {
            return true;
        }
        if (kind == RTQNAMEA) {
            return true;
        }
        if (kind == MULTINAMEL) {
            return true;
        }
        if (kind == MULTINAMELA) {
            return true;
        }
        return false;
    }

    public boolean needsName() {
        if (kind == RTQNAMEL) {
            return true;
        }
        if (kind == RTQNAMELA) {
            return true;
        }
        if (kind == MULTINAMEL) {
            return true;
        }
        if (kind == MULTINAMELA) {
            return true;
        }
        return false;
    }

    public boolean needsNs() {
        if (kind == RTQNAME) {
            return true;
        }
        if (kind == RTQNAMEA) {
            return true;
        }
        if (kind == RTQNAMEL) {
            return true;
        }
        if (kind == RTQNAMELA) {
            return true;
        }
        return false;
    }

    public String getKindStr() {
        String kindStr = "?";
        for (int k = 0; k < multinameKinds.length; k++) {
            if (multinameKinds[k] == kind) {
                kindStr = multinameKindNames[k];
                break;
            }
        }
        return kindStr;
    }

    @Override
    public String toString() {
        String kindStr = getKindStr();
        return "kind=" + kindStr + " name_index=" + name_index + " namespace_index=" + namespace_index + " namespace_set_index=" + namespace_set_index + " qname_index=" + qname_index + " params_size:" + params.size();

    }

    private static String namespaceToString(ConstantPool constants, int index) {
        if (index == 0) {
            return "null";
        }
        int type = constants.getNamespace(index).kind;
        int name_index = constants.getNamespace(index).name_index;
        String name = name_index == 0 ? null : constants.getNamespace(index).getName(constants);
        int sub = -1;
        for (int n = 1; n < constants.getNamespaceCount(); n++) {
            if (constants.getNamespace(n).kind == type && constants.getNamespace(n).hasName(constants.getNamespace(index).getName(constants), constants)) {
                sub++;
            }
            if (n == index) {
                break;
            }
        }
        return constants.getNamespace(index).getKindStr() + "(" + (name == null ? "null" : "\"" + Helper.escapeString(name) + "\"") + (sub > 0 ? ",\"" + sub + "\"" : "") + ")";
    }

    private static String namespaceSetToString(ConstantPool constants, int index) {
        if (index == 0) {
            return "null";
        }
        String ret = "[";
        for (int n = 0; n < constants.getNamespaceSet(index).namespaces.length; n++) {
            if (n > 0) {
                ret += ",";
            }
            int ns = constants.getNamespaceSet(index).namespaces[n];
            ret += namespaceToString(constants, ns);
        }
        ret += "]";
        return ret;
    }

    private static String multinameToString(ConstantPool constants, int index, List<String> fullyQualifiedNames) {
        if (index == 0) {
            return "null";
        }
        return constants.getMultiname(index).toString(constants, fullyQualifiedNames);
    }

    public String toString(ConstantPool constants, List<String> fullyQualifiedNames) {

        switch (kind) {
            case QNAME:
            case QNAMEA:
                return getKindStr() + "(" + namespaceToString(constants, namespace_index) + "," + (name_index==0?"null":"\"" + Helper.escapeString(constants.getString(name_index)) + "\"") + ")";
            case RTQNAME:
            case RTQNAMEA:
                return getKindStr() + "(" + (name_index==0?"null":"\"" + Helper.escapeString(constants.getString(name_index))) + "\"" + ")";
            case RTQNAMEL:
            case RTQNAMELA:
                return getKindStr() + "()";
            case MULTINAME:
            case MULTINAMEA:
                return getKindStr() + "(" + (name_index==0?"null":"\"" + Helper.escapeString(constants.getString(name_index)) + "\"") + "," + namespaceSetToString(constants, namespace_set_index) + ")";
            case MULTINAMEL:
            case MULTINAMELA:
                return getKindStr() + "(" + namespaceSetToString(constants, namespace_set_index) + ")";
            case TYPENAME:
                String tret = getKindStr() + "(";
                tret += multinameToString(constants, qname_index, fullyQualifiedNames);
                tret += "<";
                for (int i = 0; i < params.size(); i++) {
                    if (i > 0) {
                        tret += ",";
                    }
                    tret += multinameToString(constants, params.get(i), fullyQualifiedNames);
                }
                tret += ">";
                tret += ")";
                return tret;
        }
        return null;
    }

    private String typeNameToStr(ConstantPool constants, List<String> fullyQualifiedNames) {
        if (constants.getMultiname(qname_index).name_index == name_index) {
            return "ambiguousTypeName";
        }
        String typeNameStr = constants.getMultiname(qname_index).getName(constants, fullyQualifiedNames);
        if (!params.isEmpty()) {
            typeNameStr += ".<";
            for (int i = 0; i < params.size(); i++) {
                if (i > 0) {
                    typeNameStr += ",";
                }
                if (params.get(i) == 0) {
                    typeNameStr += "*";
                } else {
                    typeNameStr += constants.getMultiname(params.get(i)).getName(constants, fullyQualifiedNames);
                }
            }
            typeNameStr += ">";
        }
        return typeNameStr;
    }

    public String getName(ConstantPool constants, List<String> fullyQualifiedNames) {
        if (kind == TYPENAME) {
            return typeNameToStr(constants, fullyQualifiedNames);
        }
        if (name_index == -1) {
            return "";
        }
        if (name_index == 0) {
            return (isAttribute() ? "@*" : "*");
        } else {
            String name = constants.getString(name_index);
            if ((fullyQualifiedNames != null) && fullyQualifiedNames.contains(name)) {
                return getNameWithNamespace(constants);
            }
            return (isAttribute() ? "@" : "") + name;
        }
    }

    public String getNameWithNamespace(ConstantPool constants) {
        String ret = "";
        Namespace ns = getNamespace(constants);
        if (ns != null) {
            String nsname = ns.getName(constants);
            if (nsname != null && !nsname.isEmpty()) {
                ret += nsname + ".";
            }
        }
        ret += getName(constants, null);
        return ret;
    }

    public Namespace getNamespace(ConstantPool constants) {
        if ((namespace_index == 0) || (namespace_index == -1)) {
            return null;
        } else {
            return constants.getNamespace(namespace_index);
        }
    }

    public NamespaceSet getNamespaceSet(ConstantPool constants) {
        if (namespace_set_index == 0) {
            return null;
        } else if (namespace_set_index == -1) {
            return null;
        } else {
            return constants.getNamespaceSet(namespace_set_index);
        }
    }
}
